from sklearn.neighbors import KNeighborsClassifier  
  
# 定义数据和标签  
x = [[0], [1], [2], [3],[4], [5]]  
y = [0, 0, 1, 1,2,2]  
  
# 实例化k-近邻分类器（其中k=1）  
estimator = KNeighborsClassifier(n_neighbors=6)  
  
# 使用fit方法进行训练  
estimator.fit(x, y)  
  
# 使用predict方法进行预测  
prediction = estimator.predict([[9]])  
print(prediction)  # 输出预测结果